from selenium import webdriver


def open_browser(browser):
    if browser == 'chrome':
        return webdriver.Chrome()
    elif browser == 'firefox':
        return webdriver.Firefox()
    else:
        print('暂不支持该浏览器')


